from mymodel import *


model = MyModule()
# 导出为 TorchScript（ATen IR）
scripted_model = torch.jit.script(model)
print(scripted_model.code)

# 打印 IR（显示 ATen 算子）
print(scripted_model.graph)
# 输出示例：
# graph(%x.1 : Tensor, %y.1 : Tensor):
#   %2 : Tensor = aten::add(%x.1, %y.1, %1)
#   return (%2)
